# -*- coding:utf8 -*-
"""
词向量分类模型
"""
from __future__ import print_function, unicode_literals
from cover_api.w2v import 


class W2vClassify(object):
    """
    采用逻辑回归以及词向量进行分类模型
    """

    def __int__(self, model, filter, path):
        self.model = model
        self.filter = filter
        self.path = path

    def load(self, path=None):
        """
        加载模型
        :param path:
        :return:
        """
        if path is None:
            path = self.path

        self.model.load(path)

    def save(self, path=None):
        """
        保存模型
        :param path:
        :return:
        """
        if path is None:
            path = self.path

        self.model.save(path)

    def fit(self, words_list):
        """
        进行简单的计算
        :param words_list:
        :return:
        """